在访问别人的网站 的时候,有发现,一个资源某段时间内只让访问500次
那么,可以怎么实现了?
刚好看到了python 生成器,那么咱们利用生成器来,实现一下
from flask import Flask, abort
app = Flask(__name__)
def access_restriction(a=100):
for i in range(a):
if i == a:
yield('你是第{}个,访问本页面的人'.format(i+1))
else:
yield ('你是第{}个,访问本页面的人,<h1>限流策略,仅能展示10次</h1>'.format(i+1))
a = access_restriction(10)
@app.route('/')
def index():
try:
return next(a)
except StopIteration:
print('访问量达到上限')
abort(403, '访问量达到上限')